import { EditOutlined, WarningFilled } from '@ant-design/icons';
import { isQueryPresent, TimeDimensionBase } from '@cubejs-client/core';
import { AvailableMembers } from '@cubejs-client/react';
import {
  Alert,
  Button,
  Divider,
  Input,
  notification,
  Space,
  Tabs,
  Typography,
  Skeleton,
  Radio,
} from 'antd';
import deepEquals from 'fast-deep-equal';
import { useMemo, useRef, useState } from 'react';
import styled from 'styled-components';
import { CodeSnippet } from '../atoms';
import { Box, Flex } from '../grid';
import { useDeepEffect, useIsMounted, useToken } from '../hooks';
import { useDeepMemo } from '../hooks/deep-memo';
import { useCloud } from '../cloud';
import { QueryMemberKey, SchemaFormat } from '../types';
import { prettifyObject } from '../utils';
import { Cubes } from './components/Cubes';
import { Members } from './components/Members';
import { RollupSettings, Settings } from './components/Settings';
import { TimeDimension } from './components/TimeDimension';
import { useRollupDesignerContext } from './Context';
import {
  areReferencesEmpty,
  buildSettings,
  getPreAggregationReferences,
  getRollupDefinitionFromReferences,
  PreAggregationReferences,
  updateQuery,
} from './utils';
import { request } from '../shared/request';
import { getNameMemberPairs } from '../shared/members';
import { FatalError } from '../components/Error/FatalError';

const { Paragraph, Text } = Typography;
const { TabPane } = Tabs;

const Wrapper = styled.div`
  display: grid;
  grid-template-columns: 1fr 400px;
  gap: 16px;
`;

const MainBox = styled(Box)`
  & .ant-tabs-nav {
    padding-left: 24px;
    margin: 0;
  }
`;

const RollupQueryBox = styled.div`
  padding: 0 24px 32px;
  background: #f6f6f8;

  & .ant-tabs-nav {
    margin-bottom: 24px;
  }

  & .ant-tabs-tab {
    div,
    .ant-typography {
      font-weight: 500;
    }
  }

  & .ant-typography > .anticon {
    padding-left: 12px;
  }
`;

function getSelectedKeys(references: PreAggregationReferences) {
  const keys = new Set<string>();

  ['measures', 'dimensions', 'timeDimensions', 'segments'].map((memberKey) => {
    if (memberKey === 'timeDimensions') {
      const { dimension } = references[memberKey]?.[0] || {};

      if (dimension) {
        keys.add(`td:${dimension}`);
      }
    } else {
      references[memberKey]?.map((key) => key != null && keys.add(key));
    }
  });

  return Array.from(keys.values());
}

type RollupDesignerProps = {
  apiUrl: string;
  memberTypeCubeMap: AvailableMembers;
  token?: string;
};

export function RollupDesigner({
  apiUrl,
  memberTypeCubeMap,
  token: designerToken,
}: RollupDesignerProps) {
  const initialMatching = useRef<boolean>();
  const isMounted = useIsMounted();
  const appToken = useToken();

  const token = appToken || designerToken;

  const { isCloud, ...cloud } = useCloud();
  const { query, transformedQuery, isLoading, error, toggleModal, defaultSchemaFormat } =
    useRollupDesignerContext();

  const [isCronValid, setCronValidity] = useState<boolean>(true);
  const [settings, setSettings] = useState<RollupSettings>({});
  const [openKeys, setOpenKeys] = useState<string[]>([]);
  const [firstOpenCubeName, setFirstOpenCubeName] = useState<string | null>(
    null
  );

  const [matching, setMatching] = useState<boolean | undefined>();
  const [saving, setSaving] = useState<boolean>(false);
  const [preAggName, setPreAggName] = useState<string>('main');
  const [nonAdditiveMeasure, setNonAdditiveMeasure] = useState<string | null>(
    null
  );

  const [schemaFormat, setSchemaFormat] = useState<SchemaFormat>(
    defaultSchemaFormat
  );

  const { order, limit, filters, ...matchedQuery } = query || {};

  const segments = new Set<string>();
  memberTypeCubeMap.segments.forEach(({ members }) => {
    members.forEach(({ name }) => segments.add(name));
  });

  const [references, setReferences] = useState<PreAggregationReferences>(
    getPreAggregationReferences(transformedQuery, segments)
  );

  const hideMatchRollupButton = useDeepMemo(() => {
    if (matching === undefined || !initialMatching.current) {
      return true;
    }

    return deepEquals(
      references,
      getPreAggregationReferences(transformedQuery, segments)
    );
  }, [matching, transformedQuery, references, segments, initialMatching]);

  useDeepEffect(() => {
    const references = getPreAggregationReferences(transformedQuery, segments);

    setReferences(references);

    const openKeys = getSelectedKeys(references).map(
      (key) => key.split('.')[0]
    );
    setOpenKeys(openKeys);
    setFirstOpenCubeName(openKeys[0] || null);

    if (transformedQuery?.measureToLeafMeasures != null) {
      for (const [measure, leafMeasures] of Object.entries(
        transformedQuery.measureToLeafMeasures
      )) {
        if (leafMeasures.length > 1) {
          setNonAdditiveMeasure(measure);
          break;
        }
      }
    }
  }, [transformedQuery, segments]);

  const selectedKeys = useDeepMemo(() => {
    const selectedKeys = getSelectedKeys(references);

    return selectedKeys;
  }, [references]);

  useDeepEffect(() => {
    let active = true;
    const { measures, segments, dimensions, timeDimensions } = references;

    async function load() {
      const { json } = await request(
        `${apiUrl}/pre-aggregations/can-use`,
        'POST',
        {
          token: token!,
          body: {
            transformedQuery,
            references: {
              measures,
              dimensions: dimensions.concat(segments),
              timeDimensions,
            },
          },
        }
      );

      if (isMounted() && active) {
        setMatching((prevMatching) => {
          if (prevMatching === undefined) {
            initialMatching.current =
              json.canUsePreAggregationForTransformedQuery;
          }
          return json.canUsePreAggregationForTransformedQuery;
        });
      }
    }

    if (token != null && transformedQuery) {
      load();
    }

    return () => (active = false);
  }, [isMounted, references, token, transformedQuery]);

  const cubeName = useMemo(() => {
    let cubeName: string | null = null;

    if (transformedQuery) {
      cubeName = (
        transformedQuery?.leafMeasures[0] ||
        transformedQuery?.ownedDimensions[0] ||
        'CubeName'
      ).split('.')[0];
    } else if (!areReferencesEmpty(references)) {
      const [key] = getSelectedKeys(references);
      cubeName = key.split('.')[0] || null;
    }

    return cubeName;
  }, [transformedQuery, references]);

  const [
    showDecomposedMeasureAlert,
    showNonAdditiveMeasureAlert,
    showCountDistinctAlert,
  ] = useDeepMemo(() => {
    const { measureToLeafMeasures = {} } = transformedQuery || {};

    let showDecomposedMeasureAlert = false;
    let showNonAdditiveMeasureAlert = false;
    let showCountDistinctAlert = false;

    if (nonAdditiveMeasure && measureToLeafMeasures[nonAdditiveMeasure]) {
      showDecomposedMeasureAlert = measureToLeafMeasures[
        nonAdditiveMeasure
      ].every(({ additive }) => additive);
    }

    if (transformedQuery && !transformedQuery.leafMeasureAdditive) {
      const allLeafMeasures = Object.values(measureToLeafMeasures).reduce(
        (memo, leafMeasures) => [...memo, ...leafMeasures],
        []
      );

      showNonAdditiveMeasureAlert = references.measures.some((measure) => {
        const leafMeasure = allLeafMeasures.find(
          (leafMeasure) => leafMeasure.measure === measure
        );

        if (!leafMeasure) {
          return false;
        }

        return !leafMeasure.additive;
      });

      const hasCountDistinctMeasures = references.measures.some((measure) => {
        const leafMeasure = allLeafMeasures.find(
          (leafMeasure) => leafMeasure.measure === measure
        );

        if (!leafMeasure) {
          return false;
        }

        return leafMeasure.type === 'countDistinct';
      });

      showCountDistinctAlert =
        hasCountDistinctMeasures && !references.timeDimensions[0]?.granularity;
    }

    return [
      showDecomposedMeasureAlert,
      showNonAdditiveMeasureAlert,
      showCountDistinctAlert,
    ];
  }, [references, transformedQuery, nonAdditiveMeasure]);

  const indexedMembers = Object.fromEntries(
    getNameMemberPairs([
      ...memberTypeCubeMap.measures,
      ...memberTypeCubeMap.dimensions,
      ...memberTypeCubeMap.timeDimensions,
      ...memberTypeCubeMap.segments,
    ])
  );

  async function handleAddToSchemaClick() {
    if (!cubeName) {
      return;
    }

    const definition = {
      preAggregationName: preAggName,
      cubeName,
      code: getRollupDefinitionFromReferences(references, preAggName, settings, schemaFormat)
        .value,
    };

    function showSuccessMessage() {
      notification.success({
        message: `Rollup has been added to the ${cubeName} cube`,
      });
    }

    setSaving(true);

    if (!isCloud) {
      const response = await request(
        '/playground/schema/pre-aggregation',
        'POST',
        {
          body: definition,
        }
      );

      if (response.ok) {
        showSuccessMessage();
        toggleModal();
      } else {
        const { error } = response.json;
        notification.error({
          message: error,
        });
      }
    } else {
      if (cloud.addPreAggregationToSchema == null) {
        throw new Error('cloud.addPreAggregationToSchema is not defined');
      }

      const { error } = await cloud.addPreAggregationToSchema(definition);
      if (!error) {
        showSuccessMessage();
        toggleModal();
      } else {
        notification.error({
          message: error,
        });
      }
    }

    setSaving(false);
  }

  function handleMemberToggle(memberType: QueryMemberKey) {
    return (key) => {
      setReferences(updateQuery(references, memberType, key) as any);
    };
  }

  function rollupBody() {
    return (
      <>
        <CodeSnippet
          style={{ marginBottom: 16 }}
          code={
            getRollupDefinitionFromReferences(references, preAggName, settings, schemaFormat)
              .code
          }
          copyMessage="Rollup definition is copied"
          theme="light"
        />

        {cubeName ? (
          <Space direction="vertical" style={{width: '100%'}} size={32}>
            <Flex justifyContent="end" gap={2} alignItems="center">
              <Box>
                <Typography.Text>Data Model Format</Typography.Text>
              </Box>

              <Box>
                <Radio.Group
                  options={[
                    {
                      label: 'JavaScript',
                      value: SchemaFormat.js,
                    },
                    {
                      label: 'YAML',
                      value: SchemaFormat.yaml,
                    },
                  ]}
                  value={schemaFormat}
                  optionType="button"
                  buttonStyle="solid"
                  onChange={(e) => setSchemaFormat(e.target.value)}
                />
              </Box>
            </Flex>

            <Button
              data-testid="rd-add-btn"
              type="primary"
              loading={saving}
              disabled={!isCronValid}
              style={{ width: '100%' }}
              onClick={handleAddToSchemaClick}
            >
              Add to the Data Schema
            </Button>
          </Space>
        ) : null}
      </>
    );
  }

  if (isLoading) {
    return (
      <Flex gap={4} style={{ margin: 24 }}>
        <Box style={{ minWidth: 256 }}>
          <Skeleton />
        </Box>

        <Box grow={1}>
          <Skeleton />
          <Skeleton />
        </Box>

        <Box
          style={{
            width: 420,
            minWidth: 420,
          }}
        >
          <Skeleton />
        </Box>
      </Flex>
    );
  }

  if (error) {
    return (
      <div style={{ margin: '32px 24px' }}>
        <FatalError error={error} />
      </div>
    );
  }

  return (
    <Wrapper>
      <MainBox grow={1} style={{ overflowX: 'scroll' }}>
        <Tabs style={{ minHeight: '100%' }}>
          <TabPane
            tab={<span data-testid="rd-members-tab">Members</span>}
            key="members"
          >
            <Flex gap={2}>
              <Box
                style={{
                  minWidth: 280,
                  maxWidth: 280,
                }}
              >
                <Cubes
                  openKeys={openKeys}
                  selectedKeys={selectedKeys}
                  memberTypeCubeMap={memberTypeCubeMap}
                  firstOpenCubeName={firstOpenCubeName}
                  onSelect={(memberType, key) => {
                    handleMemberToggle(memberType)(key);
                  }}
                  onOpenKeysChange={setOpenKeys}
                />
              </Box>

              <Box
                grow={1}
                style={{
                  marginTop: 24,
                  overflowX: 'scroll',
                }}
              >
                {references.measures?.length ? (
                  <>
                    <Members
                      title="Measures"
                      members={references.measures.map(
                        (name) => indexedMembers[name]
                      )}
                      onRemove={handleMemberToggle('measures')}
                    />

                    <Divider />
                  </>
                ) : null}

                {references.dimensions?.length ? (
                  <>
                    <Members
                      title="Dimensions"
                      members={references.dimensions.map(
                        (name) => indexedMembers[name]
                      )}
                      onRemove={handleMemberToggle('dimensions')}
                    />

                    <Divider />
                  </>
                ) : null}

                {references.segments.length ? (
                  <>
                    <Members
                      title="Segments"
                      members={references.segments.map(
                        (name) => indexedMembers[name]
                      )}
                      onRemove={handleMemberToggle('segments')}
                    />

                    <Divider />
                  </>
                ) : null}

                {references.timeDimensions.length ? (
                  <TimeDimension
                    member={
                      indexedMembers[references.timeDimensions[0].dimension]
                    }
                    granularity={references.timeDimensions[0].granularity}
                    onGranularityChange={(granularity) => {
                      setReferences({
                        ...references,
                        timeDimensions: [
                          {
                            ...(references.timeDimensions[0] || {}),
                            ...(granularity ? { granularity } : null),
                          } as TimeDimensionBase,
                        ],
                      });
                    }}
                    onRemove={handleMemberToggle('timeDimensions')}
                  />
                ) : null}
              </Box>
            </Flex>
          </TabPane>

          <TabPane
            tab={<span data-testid="rd-settings-tab">Settings</span>}
            key="settings"
          >
            <Settings
              hasTimeDimension={references.timeDimensions.length > 0}
              members={references.measures
                .concat(references.dimensions)
                .concat(references.timeDimensions.map((td) => td.dimension))}
              onCronExpressionValidityChange={setCronValidity}
              onChange={(values) => setSettings(buildSettings(values))}
            />
          </TabPane>
        </Tabs>
      </MainBox>

      <RollupQueryBox>
        <Tabs>
          <TabPane
            tab={<span data-testid="rd-definition-tab">Rollup Definition</span>}
            key="rollup"
          >
            <Flex direction="column" justifyContent="flex-start">
              {showNonAdditiveMeasureAlert && (
                <Box style={{ marginBottom: 24 }}>
                  <Alert
                    type="info"
                    message={
                      <Text>This rollup contains a non-additive measure</Text>
                    }
                  />
                </Box>
              )}

              {showDecomposedMeasureAlert && (
                <Box style={{ marginBottom: 24 }}>
                  <Alert
                    type="info"
                    message={
                      <Text>
                        Because <b>{nonAdditiveMeasure}</b> is a non-additive
                        measure that is calculated with additive measures, this
                        rollup is configured with the additive measures that
                        calculate this non-additive measure. See more info in{' '}
                        <Typography.Link
                          href="https://cube.dev/docs/recipes/non-additivity#data-schema-decomposing-into-a-formula-with-additive-measures"
                          target="_blank"
                        >
                          our docs
                        </Typography.Link>
                        .
                      </Text>
                    }
                  />
                </Box>
              )}

              {!areReferencesEmpty(references) &&
                !references.timeDimensions.length && (
                  <Box style={{ marginBottom: 24 }}>
                    <Alert
                      type="warning"
                      message="This rollup has no time dimension so it cannot be partitioned"
                    />
                  </Box>
                )}

              <Box style={{ marginBottom: 16 }}>
                {!areReferencesEmpty(references) ? (
                  <Paragraph>
                    Add the following rollup pre-aggregation
                    <br /> to the <b>{cubeName}</b> cube:
                  </Paragraph>
                ) : (
                  <Alert type="warning" message="Add some references" />
                )}

                <Paragraph style={{ margin: '24px 0 4px' }}>
                  Rollup Name
                </Paragraph>

                <Input
                  value={preAggName}
                  suffix={<EditOutlined />}
                  onChange={(event) => setPreAggName(event.target.value)}
                />
              </Box>

              <Box>{rollupBody()}</Box>
            </Flex>
          </TabPane>

          {isQueryPresent(query) ? (
            <TabPane
              tab={
                matching ? (
                  <span data-testid="rd-query-tab">Query Compatibility</span>
                ) : (
                  <Typography.Text data-testid="rd-query-tab">
                    Query Compatibility
                    {matching != null ? (
                      <WarningFilled style={{ color: '#FBBC05' }} />
                    ) : null}
                  </Typography.Text>
                )
              }
              key="query"
            >
              <Flex direction="column" justifyContent="flex-start">
                <Box style={{ marginBottom: 32 }}>
                  <Space direction="vertical" size={24}>
                    {showCountDistinctAlert && (
                      <Alert
                        type="warning"
                        message={
                          <Text>
                            This query does not have any time dimension
                            granularity, which prevents pre-aggregating any
                            count distinct measures.
                          </Text>
                        }
                      />
                    )}

                    {matching ? (
                      <Text>This rollup will match the following query:</Text>
                    ) : (
                      <>
                        <Alert
                          data-testid="rd-incompatible-query"
                          type="warning"
                          message={
                            <Text>
                              This rollup does <b>NOT</b> match the following
                              query:
                            </Text>
                          }
                        />

                        {!hideMatchRollupButton && (
                          <Button
                            data-testid="rd-match-rollup-btn"
                            type="primary"
                            ghost
                            onClick={() => {
                              setReferences(
                                getPreAggregationReferences(
                                  transformedQuery,
                                  segments
                                )
                              );
                              setMatching(true);
                            }}
                          >
                            Match Rollup
                          </Button>
                        )}
                      </>
                    )}
                  </Space>
                </Box>

                <CodeSnippet
                  style={{ minWidth: 200 }}
                  code={prettifyObject(matchedQuery)}
                  copyMessage="Query is copied"
                  theme="light"
                />
              </Flex>
            </TabPane>
          ) : null}
        </Tabs>
      </RollupQueryBox>
    </Wrapper>
  );
}
